DSWorld: A Data Science World Model for Efficient Autonomous Agents
DSWorld:预测数据科学操作效果的世界模型框架
🧭 数据科学世界模型 · 代理训练加速
The Hong Kong University of Science and Technology (Guangzhou)
提出数据科学世界模型预测操作结果,实现代理训练加速14倍、推理加速3-6倍
📌 工作参考价值(4 点)
- 可复用的世界模型思路:在搜索推荐系统的自动化调参或特征工程代理中,借鉴“低代价操作真执行,高代价操作用LLM模拟”的策略,用历史日志训练一个轻量过渡预测模型,减少线上实验成本。
- 结构化状态表示:将工作流状态(数据Schema、模型性能、资源消耗等)组织成结构化文本输入LLM,而非原始代码或日志,能稳定提升预测准确率,可直接用于推荐agent的状态管理。
- 成本感知路由:设定成本阈值,简单操作(如数据抽样、缺失值填充)直接执行,复杂操作(如模型训练)用世界模型模拟,可构建分级探索的AutoML代理,降低计算开销。
- 反思式世界模型优化:利用真实执行后的错误反馈更新世界模型,可将自动化A/B测试的模拟器与线上结果对齐,逐步提升模拟保真度,减少策略上线前的风险评估偏差。